# Copyright (C) 2019  ABRT Team
# Copyright (C) 2019  Red Hat, Inc.
#
# This file is part of faf.
#
# faf is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# faf is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with faf.  If not, see <http://www.gnu.org/licenses/>.

"""
add TZ info to periodictasks

Revision ID: bb2289ffb392
Revises: 9596a0f03838
Create Date: 2019-12-12 14:59:44.743113
"""

from alembic.op import alter_column
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = 'bb2289ffb392'
down_revision = '9596a0f03838'


def upgrade() -> None:
    alter_column('periodictasks', 'last_run_at', type_=sa.DateTime(timezone=True), nullable=True)
    alter_column('taskresult', 'finished_time', type_=sa.DateTime(timezone=True), nullable=True)


def downgrade() -> None:
    alter_column('periodictasks', 'last_run_at', type_=sa.DateTime(timezone=False), nullable=True)
    alter_column('taskresult', 'finished_time', type_=sa.DateTime(timezone=False), nullable=True)
